from app.graph.state import State


def edge_after_intent_recog(state: State):
    intent = state["intent"]
    if intent == "weather":
        return "task_weather"
    return "task_chatbot"
